package com.sky.utils;

import com.aliyun.oss.ClientBuilderConfiguration;
import com.aliyun.oss.OSS;
import com.aliyun.oss.OSSClientBuilder;
import com.aliyun.oss.common.auth.CredentialsProviderFactory;
import com.aliyun.oss.common.auth.EnvironmentVariableCredentialsProvider;
import com.aliyun.oss.common.comm.SignVersion;
import com.sky.properties.AliyunOSSProperties;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Component;

import java.io.File;
import java.io.FileInputStream;
import java.io.InputStream;
import java.util.UUID;

/**
 * @version: 1.0
 * @Author: 李佳毅
 * @description:
 * @date: 2025-06-27 20:06
 */
@Component
public class AliyunOSSUtils {

    @Autowired
    private AliyunOSSProperties aliyunOSSProperties;

/*    @Value("${aliyun.oss.endpoint}")
    private String endpoint;*/

    //写一个静态方法,负责接受文件名,并且返回一个新的文件名
    public String uploadFile(InputStream in, String originalFilename) throws Exception {
        // Endpoint以华东1（杭州）为例，填写为https://oss-cn-hangzhou.aliyuncs.com，其它Region请按实际情况填写。
        String endpoint = aliyunOSSProperties.getEndpoint();
        String bucketName = aliyunOSSProperties.getBucketName();
        // 填写Bucket所在地域。以华东1（杭州）为例，Region填写为cn-hangzhou。
        String region = aliyunOSSProperties.getRegion();

        // 从环境变量中获取访问凭证。运行本代码示例之前，请先配置环境变量
        EnvironmentVariableCredentialsProvider credentialsProvider =
                CredentialsProviderFactory.newEnvironmentVariableCredentialsProvider();

        // 创建OSSClient实例。
        // 当OSSClient实例不再使用时，调用shutdown方法以释放资源。
        ClientBuilderConfiguration clientBuilderConfiguration = new ClientBuilderConfiguration();
        // 显式声明使用 V4 签名算法
        clientBuilderConfiguration.setSignatureVersion(SignVersion.V4);
        OSS ossClient = OSSClientBuilder.create()
                .endpoint(endpoint)
                .credentialsProvider(credentialsProvider)
                .region(region)
                .build();
        //TODO 这里写代码逻辑
        //使用UUID + 原文件名 得到新的文件名
        String newName = UUID.randomUUID().toString() + "-" + originalFilename;
        //然后根据新文件名得到目录名字
        String newDirName = getNewDirName(newName);
        //最终上传的文件名是 目录+文件名的格式 目录阿里云OSS会自动创建
        String finalName = newDirName + newName;
        try {
            ossClient.putObject(bucketName, finalName, in);
            System.out.println("2. 文件 " + finalName + " 上传成功。");
        } finally {
            if (ossClient != null) {
                ossClient.shutdown();
            }
        }
        return "https://" + bucketName + "." + endpoint.split("//")[1] + "/" + finalName;
    }

    //写一个静态方法,用来生成新的目录名称
    private static String getNewDirName(String fileName) {
        StringBuilder sb = new StringBuilder();
        int hash = fileName.hashCode();
        for (int i = 0; i < 8; i++) {
            int subDir = hash & 15;//得到目录的子目录
            sb.append(subDir + "/");
            //进行位运算
            hash >>= 4;
        }
        return sb.toString();
    }

    //上传普通文件
    public String uploadOrdinaryFile(File file) throws Exception {
        String originalFilename = "aaa.txt";
        InputStream in = new FileInputStream(file);
        // Endpoint以华东1（杭州）为例，填写为https://oss-cn-hangzhou.aliyuncs.com，其它Region请按实际情况填写。
        String endpoint = aliyunOSSProperties.getEndpoint();
        String bucketName = aliyunOSSProperties.getBucketName();
        // 填写Bucket所在地域。以华东1（杭州）为例，Region填写为cn-hangzhou。
        String region = aliyunOSSProperties.getRegion();

        // 从环境变量中获取访问凭证。运行本代码示例之前，请先配置环境变量
        EnvironmentVariableCredentialsProvider credentialsProvider =
                CredentialsProviderFactory.newEnvironmentVariableCredentialsProvider();

        // 创建OSSClient实例。
        // 当OSSClient实例不再使用时，调用shutdown方法以释放资源。
        ClientBuilderConfiguration clientBuilderConfiguration = new ClientBuilderConfiguration();
        // 显式声明使用 V4 签名算法
        clientBuilderConfiguration.setSignatureVersion(SignVersion.V4);
        OSS ossClient = OSSClientBuilder.create()
                .endpoint(endpoint)
                .credentialsProvider(credentialsProvider)
                .region(region)
                .build();
        //TODO 这里写代码逻辑
        //使用UUID + 原文件名 得到新的文件名
        String newName = UUID.randomUUID().toString() + "-" + originalFilename;
        //然后根据新文件名得到目录名字
        String newDirName = getNewDirName(newName);
        //最终上传的文件名是 目录+文件名的格式 目录阿里云OSS会自动创建
        String finalName = newDirName + newName;
        try {
            ossClient.putObject(bucketName, finalName, in);
            System.out.println("2. 文件 " + finalName + " 上传成功。");
        } finally {
            if (ossClient != null) {
                ossClient.shutdown();
            }
        }
        return "https://" + bucketName + "." + endpoint.split("//")[1] + "/" + finalName;
    }

}
